Why am I getting IndexError: too many indices for array in my NumPy code